package top.shiyiri.servlet;

import org.apache.commons.io.IOUtils;
import sun.misc.BASE64Encoder;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;

/**
 * @author Aunean
 * @date 2022/2/1 11:59
 */
public class DownloadServlet extends HttpServlet  {

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //1.获取要下载的文件名
        String downloadFileName = "g.jpg";

        //2.读取要下载的文件内容（通过 ServletContext 对象可以读取）
        ServletContext servletContext = getServletContext();
        String mimeType = servletContext.getMimeType("/file/" + downloadFileName);
        System.out.println("下载的文件类型：" + mimeType);
        //3.在回传前，通过响应头告诉客户端返回的数据类型
        resp.setContentType(mimeType);
        //4.告诉客户端收到的数据是否用于下载使用（使用响应头）
        /*
            Content-Disposition：响应头，表示接收到的数据怎么处理
            attachment：表示附件，下载使用
            filename=文件名：表示指定下载的文件名
         */
        if (req.getHeader("User-Agent").contains("Firefox")) {
            //如果是火狐浏览器使用Base64编码
            resp.setHeader("Content-Disposition", "attachment; filename==?UTF-8?B?" +
                    new BASE64Encoder().encode("表情包.jpg".getBytes(StandardCharsets.UTF_8)) + "?=");
        } else {
            //如果不是火狐浏览器使用URL编码操作
            resp.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode("表情包.jpg", "UTF-8"));
        }

        //resp.setHeader("Content-Disposition", "attachment; filename=" + downloadFileName);

        /*
            / 斜杆被服务器解析表示地址为 http://ip:port/工程名/ 映射到代码的 web 目录
         */
        InputStream resourceAsStream = servletContext.getResourceAsStream("/file/" + downloadFileName);
        //获取响应的输出流
        OutputStream outputStream = resp.getOutputStream();

        //5.把下载的文件内容回传给客户端
        //读取输入流中全部的数据，复制给客户端，输出给客户端
        IOUtils.copy(resourceAsStream, outputStream);

    }
}
